package 动态规划;

public class 最佳买卖股票时机含冷冻期 {
//    题解：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/solution/zui-jia-mai-mai-gu-piao-shi-ji-han-leng-dong-qi-4/
    public int maxProfit(int[] prices) {
        int n = prices.length;
//我们目前持有一支股票，对应的「累计最大收益」记为 f[i][0]
//我们目前不持有任何股票，并且处于冷冻期中，对应的「累计最大收益」记为 f[i][1]
//我们目前不持有任何股票，并且不处于冷冻期中，对应的「累计最大收益」记为 f[i][2]
        int[][] dp = new int[n][3];
        dp[0][0] = -prices[0];
        for (int i = 1; i < n; i++) {
            dp[i][0] = Math.max(dp[i-1][0],dp[i-1][2]-prices[i]);
            dp[i][1] = dp[i-1][0] + prices[i];
            dp[i][2] = Math.max(dp[i-1][2],dp[i-1][1]);
        }
        return Math.max(dp[n-1][1],dp[n-1][2]);
    }
}
